package com;

import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.generator.BaseModelGenerator;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.activerecord.generator.ModelGenerator;
import com.jfinal.plugin.druid.DruidPlugin;

import javax.sql.DataSource;

public class App {

    public static String GeneratorModel(){
        String rootPath = System.getProperty("user.dir");

        // base model 所使用的包名
        String baseModelPackageName = PropKit.get("basePath")+".model.base";
        // base model 文件保存路径

        String baseModelOutputDir = rootPath + "/"+PropKit.get("srcFolder")+"/"+FileNameUtil.toPathByPackage(PropKit.get("basePath"))+"/model/base";

        // model 所使用的包名 (MappingKit 默认使用的包名)
        String modelPackageName = PropKit.get("basePath")+".model";
        // model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
        String modelOutputDir = rootPath + "/"+PropKit.get("srcFolder")+"/"+FileNameUtil.toPathByPackage(PropKit.get("basePath"))+"/model";

        BaseModelGenerator baseGe = new BaseModelGenerator(baseModelPackageName, baseModelOutputDir);
        ModelGenerator modelGe = new ModelGenerator(modelPackageName, baseModelPackageName, modelOutputDir);
        Generator gernerator = new Generator(getDataSource(), baseGe, modelGe);
        // 设置是否在 Model 中生成 dao 对象
        gernerator.setGenerateDaoInModel(true);
        // 设置是否生成字典文件
        gernerator.setGenerateDataDictionary(false);
        // 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
        gernerator.setRemovedTableNamePrefixes("t_");
        gernerator.addExcludedTable("activities_view");
        gernerator.generate();
        return modelOutputDir;
    }

    public static DataSource getDataSource() {
        DruidPlugin druidPlugin = new DruidPlugin(PropKit.get("jdbc.url"), PropKit.get("jdbc.username"), PropKit.get("jdbc.password").trim());
        //获取table remarks信息
        druidPlugin.setConnectionProperties("useInformationSchema=true;remarks=true");
        druidPlugin.start();
        return druidPlugin.getDataSource();
    }
}

